python programming math formula